home *** CD-ROM | disk | FTP | other *** search
/ Personal Computer World 2009 February / PCWFEB09.iso / Software / Linux / Kubuntu 8.10 / kubuntu-8.10-desktop-i386.iso / casper / filesystem.squashfs / usr / lib / python2.5 / urllib2.pyc (.txt) < prev    next >
Python Compiled Bytecode  |  2008-10-29  |  44KB  |  1,353 lines

  1. # Source Generated with Decompyle++
  2. # File: in.pyc (Python 2.5)
  3.  
  4. '''An extensible library for opening URLs using a variety of protocols
  5.  
  6. The simplest way to use this module is to call the urlopen function,
  7. which accepts a string containing a URL or a Request object (described
  8. below).  It opens the URL and returns the results as file-like
  9. object; the returned object has some extra methods described below.
  10.  
  11. The OpenerDirector manages a collection of Handler objects that do
  12. all the actual work.  Each Handler implements a particular protocol or
  13. option.  The OpenerDirector is a composite object that invokes the
  14. Handlers needed to open the requested URL.  For example, the
  15. HTTPHandler performs HTTP GET and POST requests and deals with
  16. non-error returns.  The HTTPRedirectHandler automatically deals with
  17. HTTP 301, 302, 303 and 307 redirect errors, and the HTTPDigestAuthHandler
  18. deals with digest authentication.
  19.  
  20. urlopen(url, data=None) -- basic usage is the same as original
  21. urllib.  pass the url and optionally data to post to an HTTP URL, and
  22. get a file-like object back.  One difference is that you can also pass
  23. a Request instance instead of URL.  Raises a URLError (subclass of
  24. IOError); for HTTP errors, raises an HTTPError, which can also be
  25. treated as a valid response.
  26.  
  27. build_opener -- function that creates a new OpenerDirector instance.
  28. will install the default handlers.  accepts one or more Handlers as
  29. arguments, either instances or Handler classes that it will
  30. instantiate.  if one of the argument is a subclass of the default
  31. handler, the argument will be installed instead of the default.
  32.  
  33. install_opener -- installs a new opener as the default opener.
  34.  
  35. objects of interest:
  36. OpenerDirector --
  37.  
  38. Request -- an object that encapsulates the state of a request.  the
  39. state can be a simple as the URL.  it can also include extra HTTP
  40. headers, e.g. a User-Agent.
  41.  
  42. BaseHandler --
  43.  
  44. exceptions:
  45. URLError-- a subclass of IOError, individual protocols have their own
  46. specific subclass
  47.  
  48. HTTPError-- also a valid HTTP response, so you can treat an HTTP error
  49. as an exceptional event or valid response
  50.  
  51. internals:
  52. BaseHandler and parent
  53. _call_chain conventions
  54.  
  55. Example usage:
  56.  
  57. import urllib2
  58.  
  59. # set up authentication info
  60. authinfo = urllib2.HTTPBasicAuthHandler()
  61. authinfo.add_password(realm=\'PDQ Application\',
  62.                       uri=\'https://mahler:8092/site-updates.py\',
  63.                       user=\'klem\',
  64.                       passwd=\'geheim$parole\')
  65.  
  66. proxy_support = urllib2.ProxyHandler({"http" : "http://ahad-haam:3128"})
  67.  
  68. # build a new opener that adds authentication and caching FTP handlers
  69. opener = urllib2.build_opener(proxy_support, authinfo, urllib2.CacheFTPHandler)
  70.  
  71. # install it
  72. urllib2.install_opener(opener)
  73.  
  74. f = urllib2.urlopen(\'http://www.python.org/\')
  75.  
  76.  
  77. '''
  78. import base64
  79. import hashlib
  80. import httplib
  81. import mimetools
  82. import os
  83. import posixpath
  84. import random
  85. import re
  86. import socket
  87. import sys
  88. import time
  89. import urlparse
  90. import bisect
  91.  
  92. try:
  93.     from cStringIO import StringIO
  94. except ImportError:
  95.     from StringIO import StringIO
  96.  
  97. from urllib import unwrap, unquote, splittype, splithost, quote, addinfourl, splitport, splitgophertype, splitquery, splitattr, ftpwrapper, noheaders, splituser, splitpasswd, splitvalue
  98. from urllib import localhost, url2pathname, getproxies
  99. __version__ = sys.version[:3]
  100. _opener = None
  101.  
  102. def urlopen(url, data = None):
  103.     global _opener
  104.     if _opener is None:
  105.         _opener = build_opener()
  106.     
  107.     return _opener.open(url, data)
  108.  
  109.  
  110. def install_opener(opener):
  111.     global _opener
  112.     _opener = opener
  113.  
  114.  
  115. class URLError(IOError):
  116.     
  117.     def __init__(self, reason):
  118.         self.args = (reason,)
  119.         self.reason = reason
  120.  
  121.     
  122.     def __str__(self):
  123.         return '<urlopen error %s>' % self.reason
  124.  
  125.  
  126.  
  127. class HTTPError(URLError, addinfourl):
  128.     '''Raised when HTTP error occurs, but also acts like non-error return'''
  129.     __super_init = addinfourl.__init__
  130.     
  131.     def __init__(self, url, code, msg, hdrs, fp):
  132.         self.code = code
  133.         self.msg = msg
  134.         self.hdrs = hdrs
  135.         self.fp = fp
  136.         self.filename = url
  137.         if fp is not None:
  138.             self._HTTPError__super_init(fp, hdrs, url)
  139.         
  140.  
  141.     
  142.     def __str__(self):
  143.         return 'HTTP Error %s: %s' % (self.code, self.msg)
  144.  
  145.  
  146.  
  147. class GopherError(URLError):
  148.     pass
  149.  
  150. _cut_port_re = re.compile(':\\d+$')
  151.  
  152. def request_host(request):
  153.     '''Return request-host, as defined by RFC 2965.
  154.  
  155.     Variation from RFC: returned value is lowercased, for convenient
  156.     comparison.
  157.  
  158.     '''
  159.     url = request.get_full_url()
  160.     host = urlparse.urlparse(url)[1]
  161.     if host == '':
  162.         host = request.get_header('Host', '')
  163.     
  164.     host = _cut_port_re.sub('', host, 1)
  165.     return host.lower()
  166.  
  167.  
  168. class Request:
  169.     
  170.     def __init__(self, url, data = None, headers = { }, origin_req_host = None, unverifiable = False):
  171.         self._Request__original = unwrap(url)
  172.         self.type = None
  173.         self.host = None
  174.         self.port = None
  175.         self.data = data
  176.         self.headers = { }
  177.         for key, value in headers.items():
  178.             self.add_header(key, value)
  179.         
  180.         self.unredirected_hdrs = { }
  181.         if origin_req_host is None:
  182.             origin_req_host = request_host(self)
  183.         
  184.         self.origin_req_host = origin_req_host
  185.         self.unverifiable = unverifiable
  186.  
  187.     
  188.     def __getattr__(self, attr):
  189.         if attr[:12] == '_Request__r_':
  190.             name = attr[12:]
  191.             if hasattr(Request, 'get_' + name):
  192.                 getattr(self, 'get_' + name)()
  193.                 return getattr(self, attr)
  194.             
  195.         
  196.         raise AttributeError, attr
  197.  
  198.     
  199.     def get_method(self):
  200.         if self.has_data():
  201.             return 'POST'
  202.         else:
  203.             return 'GET'
  204.  
  205.     
  206.     def add_data(self, data):
  207.         self.data = data
  208.  
  209.     
  210.     def has_data(self):
  211.         return self.data is not None
  212.  
  213.     
  214.     def get_data(self):
  215.         return self.data
  216.  
  217.     
  218.     def get_full_url(self):
  219.         return self._Request__original
  220.  
  221.     
  222.     def get_type(self):
  223.         if self.type is None:
  224.             (self.type, self._Request__r_type) = splittype(self._Request__original)
  225.             if self.type is None:
  226.                 raise ValueError, 'unknown url type: %s' % self._Request__original
  227.             
  228.         
  229.         return self.type
  230.  
  231.     
  232.     def get_host(self):
  233.         if self.host is None:
  234.             (self.host, self._Request__r_host) = splithost(self._Request__r_type)
  235.             if self.host:
  236.                 self.host = unquote(self.host)
  237.             
  238.         
  239.         return self.host
  240.  
  241.     
  242.     def get_selector(self):
  243.         return self._Request__r_host
  244.  
  245.     
  246.     def set_proxy(self, host, type):
  247.         self.host = host
  248.         self.type = type
  249.         self._Request__r_host = self._Request__original
  250.  
  251.     
  252.     def get_origin_req_host(self):
  253.         return self.origin_req_host
  254.  
  255.     
  256.     def is_unverifiable(self):
  257.         return self.unverifiable
  258.  
  259.     
  260.     def add_header(self, key, val):
  261.         self.headers[key.capitalize()] = val
  262.  
  263.     
  264.     def add_unredirected_header(self, key, val):
  265.         self.unredirected_hdrs[key.capitalize()] = val
  266.  
  267.     
  268.     def has_header(self, header_name):
  269.         if not header_name in self.headers:
  270.             pass
  271.         return header_name in self.unredirected_hdrs
  272.  
  273.     
  274.     def get_header(self, header_name, default = None):
  275.         return self.headers.get(header_name, self.unredirected_hdrs.get(header_name, default))
  276.  
  277.     
  278.     def header_items(self):
  279.         hdrs = self.unredirected_hdrs.copy()
  280.         hdrs.update(self.headers)
  281.         return hdrs.items()
  282.  
  283.  
  284.  
  285. class OpenerDirector:
  286.     
  287.     def __init__(self):
  288.         client_version = 'Python-urllib/%s' % __version__
  289.         self.addheaders = [
  290.             ('User-agent', client_version)]
  291.         self.handlers = []
  292.         self.handle_open = { }
  293.         self.handle_error = { }
  294.         self.process_response = { }
  295.         self.process_request = { }
  296.  
  297.     
  298.     def add_handler(self, handler):
  299.         if not hasattr(handler, 'add_parent'):
  300.             raise TypeError('expected BaseHandler instance, got %r' % type(handler))
  301.         
  302.         added = False
  303.         for meth in dir(handler):
  304.             if meth in ('redirect_request', 'do_open', 'proxy_open'):
  305.                 continue
  306.             
  307.             i = meth.find('_')
  308.             protocol = meth[:i]
  309.             condition = meth[i + 1:]
  310.             if condition.startswith('error'):
  311.                 j = condition.find('_') + i + 1
  312.                 kind = meth[j + 1:]
  313.                 
  314.                 try:
  315.                     kind = int(kind)
  316.                 except ValueError:
  317.                     pass
  318.  
  319.                 lookup = self.handle_error.get(protocol, { })
  320.                 self.handle_error[protocol] = lookup
  321.             elif condition == 'open':
  322.                 kind = protocol
  323.                 lookup = self.handle_open
  324.             elif condition == 'response':
  325.                 kind = protocol
  326.                 lookup = self.process_response
  327.             elif condition == 'request':
  328.                 kind = protocol
  329.                 lookup = self.process_request
  330.             
  331.             handlers = lookup.setdefault(kind, [])
  332.             if handlers:
  333.                 bisect.insort(handlers, handler)
  334.             else:
  335.                 handlers.append(handler)
  336.             added = True
  337.         
  338.         if added:
  339.             bisect.insort(self.handlers, handler)
  340.             handler.add_parent(self)
  341.         
  342.  
  343.     
  344.     def close(self):
  345.         pass
  346.  
  347.     
  348.     def _call_chain(self, chain, kind, meth_name, *args):
  349.         handlers = chain.get(kind, ())
  350.         for handler in handlers:
  351.             func = getattr(handler, meth_name)
  352.             result = func(*args)
  353.             if result is not None:
  354.                 return result
  355.                 continue
  356.         
  357.  
  358.     
  359.     def open(self, fullurl, data = None):
  360.         if isinstance(fullurl, basestring):
  361.             req = Request(fullurl, data)
  362.         else:
  363.             req = fullurl
  364.             if data is not None:
  365.                 req.add_data(data)
  366.             
  367.         protocol = req.get_type()
  368.         meth_name = protocol + '_request'
  369.         for processor in self.process_request.get(protocol, []):
  370.             meth = getattr(processor, meth_name)
  371.             req = meth(req)
  372.         
  373.         response = self._open(req, data)
  374.         meth_name = protocol + '_response'
  375.         for processor in self.process_response.get(protocol, []):
  376.             meth = getattr(processor, meth_name)
  377.             response = meth(req, response)
  378.         
  379.         return response
  380.  
  381.     
  382.     def _open(self, req, data = None):
  383.         result = self._call_chain(self.handle_open, 'default', 'default_open', req)
  384.         if result:
  385.             return result
  386.         
  387.         protocol = req.get_type()
  388.         result = self._call_chain(self.handle_open, protocol, protocol + '_open', req)
  389.         if result:
  390.             return result
  391.         
  392.         return self._call_chain(self.handle_open, 'unknown', 'unknown_open', req)
  393.  
  394.     
  395.     def error(self, proto, *args):
  396.         if proto in ('http', 'https'):
  397.             dict = self.handle_error['http']
  398.             proto = args[2]
  399.             meth_name = 'http_error_%s' % proto
  400.             http_err = 1
  401.             orig_args = args
  402.         else:
  403.             dict = self.handle_error
  404.             meth_name = proto + '_error'
  405.             http_err = 0
  406.         args = (dict, proto, meth_name) + args
  407.         result = self._call_chain(*args)
  408.         if result:
  409.             return result
  410.         
  411.         if http_err:
  412.             args = (dict, 'default', 'http_error_default') + orig_args
  413.             return self._call_chain(*args)
  414.         
  415.  
  416.  
  417.  
  418. def build_opener(*handlers):
  419.     '''Create an opener object from a list of handlers.
  420.  
  421.     The opener will use several default handlers, including support
  422.     for HTTP and FTP.
  423.  
  424.     If any of the handlers passed as arguments are subclasses of the
  425.     default handlers, the default handlers will not be used.
  426.     '''
  427.     import types
  428.     
  429.     def isclass(obj):
  430.         if not isinstance(obj, types.ClassType):
  431.             pass
  432.         return hasattr(obj, '__bases__')
  433.  
  434.     opener = OpenerDirector()
  435.     default_classes = [
  436.         ProxyHandler,
  437.         UnknownHandler,
  438.         HTTPHandler,
  439.         HTTPDefaultErrorHandler,
  440.         HTTPRedirectHandler,
  441.         FTPHandler,
  442.         FileHandler,
  443.         HTTPErrorProcessor]
  444.     if hasattr(httplib, 'HTTPS'):
  445.         default_classes.append(HTTPSHandler)
  446.     
  447.     skip = set()
  448.     for klass in default_classes:
  449.         for check in handlers:
  450.             if isclass(check):
  451.                 if issubclass(check, klass):
  452.                     skip.add(klass)
  453.                 
  454.             issubclass(check, klass)
  455.             if isinstance(check, klass):
  456.                 skip.add(klass)
  457.                 continue
  458.         
  459.     
  460.     for klass in skip:
  461.         default_classes.remove(klass)
  462.     
  463.     for klass in default_classes:
  464.         opener.add_handler(klass())
  465.     
  466.     for h in handlers:
  467.         if isclass(h):
  468.             h = h()
  469.         
  470.         opener.add_handler(h)
  471.     
  472.     return opener
  473.  
  474.  
  475. class BaseHandler:
  476.     handler_order = 500
  477.     
  478.     def add_parent(self, parent):
  479.         self.parent = parent
  480.  
  481.     
  482.     def close(self):
  483.         pass
  484.  
  485.     
  486.     def __lt__(self, other):
  487.         if not hasattr(other, 'handler_order'):
  488.             return True
  489.         
  490.         return self.handler_order < other.handler_order
  491.  
  492.  
  493.  
  494. class HTTPErrorProcessor(BaseHandler):
  495.     '''Process HTTP error responses.'''
  496.     handler_order = 1000
  497.     
  498.     def http_response(self, request, response):
  499.         code = response.code
  500.         msg = response.msg
  501.         hdrs = response.info()
  502.         if code not in (200, 206):
  503.             response = self.parent.error('http', request, response, code, msg, hdrs)
  504.         
  505.         return response
  506.  
  507.     https_response = http_response
  508.  
  509.  
  510. class HTTPDefaultErrorHandler(BaseHandler):
  511.     
  512.     def http_error_default(self, req, fp, code, msg, hdrs):
  513.         raise HTTPError(req.get_full_url(), code, msg, hdrs, fp)
  514.  
  515.  
  516.  
  517. class HTTPRedirectHandler(BaseHandler):
  518.     max_repeats = 4
  519.     max_redirections = 10
  520.     
  521.     def redirect_request(self, req, fp, code, msg, headers, newurl):
  522.         """Return a Request or None in response to a redirect.
  523.  
  524.         This is called by the http_error_30x methods when a
  525.         redirection response is received.  If a redirection should
  526.         take place, return a new Request to allow http_error_30x to
  527.         perform the redirect.  Otherwise, raise HTTPError if no-one
  528.         else should try to handle this url.  Return None if you can't
  529.         but another Handler might.
  530.         """
  531.         m = req.get_method()
  532.         if (code in (301, 302, 303, 307) or m in ('GET', 'HEAD') or code in (301, 302, 303)) and m == 'POST':
  533.             newurl = newurl.replace(' ', '%20')
  534.             return Request(newurl, headers = req.headers, origin_req_host = req.get_origin_req_host(), unverifiable = True)
  535.         else:
  536.             raise HTTPError(req.get_full_url(), code, msg, headers, fp)
  537.  
  538.     
  539.     def http_error_302(self, req, fp, code, msg, headers):
  540.         if 'location' in headers:
  541.             newurl = headers.getheaders('location')[0]
  542.         elif 'uri' in headers:
  543.             newurl = headers.getheaders('uri')[0]
  544.         else:
  545.             return None
  546.         newurl = urlparse.urljoin(req.get_full_url(), newurl)
  547.         new = self.redirect_request(req, fp, code, msg, headers, newurl)
  548.         if new is None:
  549.             return None
  550.         
  551.         if hasattr(req, 'redirect_dict'):
  552.             visited = new.redirect_dict = req.redirect_dict
  553.             if visited.get(newurl, 0) >= self.max_repeats or len(visited) >= self.max_redirections:
  554.                 raise HTTPError(req.get_full_url(), code, self.inf_msg + msg, headers, fp)
  555.             
  556.         else:
  557.             visited = new.redirect_dict = req.redirect_dict = { }
  558.         visited[newurl] = visited.get(newurl, 0) + 1
  559.         fp.read()
  560.         fp.close()
  561.         return self.parent.open(new)
  562.  
  563.     http_error_301 = http_error_303 = http_error_307 = http_error_302
  564.     inf_msg = 'The HTTP server returned a redirect error that would lead to an infinite loop.\nThe last 30x error message was:\n'
  565.  
  566.  
  567. def _parse_proxy(proxy):
  568.     """Return (scheme, user, password, host/port) given a URL or an authority.
  569.  
  570.     If a URL is supplied, it must have an authority (host:port) component.
  571.     According to RFC 3986, having an authority component means the URL must
  572.     have two slashes after the scheme:
  573.  
  574.     >>> _parse_proxy('file:/ftp.example.com/')
  575.     Traceback (most recent call last):
  576.     ValueError: proxy URL with no authority: 'file:/ftp.example.com/'
  577.  
  578.     The first three items of the returned tuple may be None.
  579.  
  580.     Examples of authority parsing:
  581.  
  582.     >>> _parse_proxy('proxy.example.com')
  583.     (None, None, None, 'proxy.example.com')
  584.     >>> _parse_proxy('proxy.example.com:3128')
  585.     (None, None, None, 'proxy.example.com:3128')
  586.  
  587.     The authority component may optionally include userinfo (assumed to be
  588.     username:password):
  589.  
  590.     >>> _parse_proxy('joe:password@proxy.example.com')
  591.     (None, 'joe', 'password', 'proxy.example.com')
  592.     >>> _parse_proxy('joe:password@proxy.example.com:3128')
  593.     (None, 'joe', 'password', 'proxy.example.com:3128')
  594.  
  595.     Same examples, but with URLs instead:
  596.  
  597.     >>> _parse_proxy('http://proxy.example.com/')
  598.     ('http', None, None, 'proxy.example.com')
  599.     >>> _parse_proxy('http://proxy.example.com:3128/')
  600.     ('http', None, None, 'proxy.example.com:3128')
  601.     >>> _parse_proxy('http://joe:password@proxy.example.com/')
  602.     ('http', 'joe', 'password', 'proxy.example.com')
  603.     >>> _parse_proxy('http://joe:password@proxy.example.com:3128')
  604.     ('http', 'joe', 'password', 'proxy.example.com:3128')
  605.  
  606.     Everything after the authority is ignored:
  607.  
  608.     >>> _parse_proxy('ftp://joe:password@proxy.example.com/rubbish:3128')
  609.     ('ftp', 'joe', 'password', 'proxy.example.com')
  610.  
  611.     Test for no trailing '/' case:
  612.  
  613.     >>> _parse_proxy('http://joe:password@proxy.example.com')
  614.     ('http', 'joe', 'password', 'proxy.example.com')
  615.  
  616.     """
  617.     (scheme, r_scheme) = splittype(proxy)
  618.     if not r_scheme.startswith('/'):
  619.         scheme = None
  620.         authority = proxy
  621.     elif not r_scheme.startswith('//'):
  622.         raise ValueError('proxy URL with no authority: %r' % proxy)
  623.     
  624.     end = r_scheme.find('/', 2)
  625.     if end == -1:
  626.         end = None
  627.     
  628.     authority = r_scheme[2:end]
  629.     (userinfo, hostport) = splituser(authority)
  630.     if userinfo is not None:
  631.         (user, password) = splitpasswd(userinfo)
  632.     else:
  633.         user = None
  634.         password = None
  635.     return (scheme, user, password, hostport)
  636.  
  637.  
  638. class ProxyHandler(BaseHandler):
  639.     handler_order = 100
  640.     
  641.     def __init__(self, proxies = None):
  642.         if proxies is None:
  643.             proxies = getproxies()
  644.         
  645.         if not hasattr(proxies, 'has_key'):
  646.             raise AssertionError, 'proxies must be a mapping'
  647.         self.proxies = proxies
  648.         for type, url in proxies.items():
  649.             setattr(self, '%s_open' % type, (lambda r, proxy = url, type = type, meth = self.proxy_open: meth(r, proxy, type)))
  650.         
  651.  
  652.     
  653.     def proxy_open(self, req, proxy, type):
  654.         orig_type = req.get_type()
  655.         (proxy_type, user, password, hostport) = _parse_proxy(proxy)
  656.         if proxy_type is None:
  657.             proxy_type = orig_type
  658.         
  659.         if user and password:
  660.             user_pass = '%s:%s' % (unquote(user), unquote(password))
  661.             creds = base64.b64encode(user_pass).strip()
  662.             req.add_header('Proxy-authorization', 'Basic ' + creds)
  663.         
  664.         hostport = unquote(hostport)
  665.         req.set_proxy(hostport, proxy_type)
  666.         if orig_type == proxy_type:
  667.             return None
  668.         else:
  669.             return self.parent.open(req)
  670.  
  671.  
  672.  
  673. class HTTPPasswordMgr:
  674.     
  675.     def __init__(self):
  676.         self.passwd = { }
  677.  
  678.     
  679.     def add_password(self, realm, uri, user, passwd):
  680.         if isinstance(uri, basestring):
  681.             uri = [
  682.                 uri]
  683.         
  684.         if realm not in self.passwd:
  685.             self.passwd[realm] = { }
  686.         
  687.         for default_port in (True, False):
  688.             reduced_uri = []([ self.reduce_uri(u, default_port) for u in uri ])
  689.             self.passwd[realm][reduced_uri] = (user, passwd)
  690.         
  691.  
  692.     
  693.     def find_user_password(self, realm, authuri):
  694.         domains = self.passwd.get(realm, { })
  695.         for default_port in (True, False):
  696.             reduced_authuri = self.reduce_uri(authuri, default_port)
  697.             for uris, authinfo in domains.iteritems():
  698.                 for uri in uris:
  699.                     if self.is_suburi(uri, reduced_authuri):
  700.                         return authinfo
  701.                         continue
  702.                 
  703.             
  704.         
  705.         return (None, None)
  706.  
  707.     
  708.     def reduce_uri(self, uri, default_port = True):
  709.         '''Accept authority or URI and extract only the authority and path.'''
  710.         parts = urlparse.urlsplit(uri)
  711.         if parts[1]:
  712.             scheme = parts[0]
  713.             authority = parts[1]
  714.             if not parts[2]:
  715.                 pass
  716.             path = '/'
  717.         else:
  718.             scheme = None
  719.             authority = uri
  720.             path = '/'
  721.         (host, port) = splitport(authority)
  722.         if default_port and port is None and scheme is not None:
  723.             dport = {
  724.                 'http': 80,
  725.                 'https': 443 }.get(scheme)
  726.             if dport is not None:
  727.                 authority = '%s:%d' % (host, dport)
  728.             
  729.         
  730.         return (authority, path)
  731.  
  732.     
  733.     def is_suburi(self, base, test):
  734.         '''Check if test is below base in a URI tree
  735.  
  736.         Both args must be URIs in reduced form.
  737.         '''
  738.         if base == test:
  739.             return True
  740.         
  741.         if base[0] != test[0]:
  742.             return False
  743.         
  744.         common = posixpath.commonprefix((base[1], test[1]))
  745.         if len(common) == len(base[1]):
  746.             return True
  747.         
  748.         return False
  749.  
  750.  
  751.  
  752. class HTTPPasswordMgrWithDefaultRealm(HTTPPasswordMgr):
  753.     
  754.     def find_user_password(self, realm, authuri):
  755.         (user, password) = HTTPPasswordMgr.find_user_password(self, realm, authuri)
  756.         if user is not None:
  757.             return (user, password)
  758.         
  759.         return HTTPPasswordMgr.find_user_password(self, None, authuri)
  760.  
  761.  
  762.  
  763. class AbstractBasicAuthHandler:
  764.     rx = re.compile('(?:.*,)*[ \t]*([^ \t]+)[ \t]+realm="([^"]*)"', re.I)
  765.     
  766.     def __init__(self, password_mgr = None):
  767.         if password_mgr is None:
  768.             password_mgr = HTTPPasswordMgr()
  769.         
  770.         self.passwd = password_mgr
  771.         self.add_password = self.passwd.add_password
  772.  
  773.     
  774.     def http_error_auth_reqed(self, authreq, host, req, headers):
  775.         authreq = headers.get(authreq, None)
  776.         if authreq:
  777.             mo = AbstractBasicAuthHandler.rx.search(authreq)
  778.             if mo:
  779.                 (scheme, realm) = mo.groups()
  780.                 if scheme.lower() == 'basic':
  781.                     return self.retry_http_basic_auth(host, req, realm)
  782.                 
  783.             
  784.         
  785.  
  786.     
  787.     def retry_http_basic_auth(self, host, req, realm):
  788.         (user, pw) = self.passwd.find_user_password(realm, host)
  789.         if pw is not None:
  790.             raw = '%s:%s' % (user, pw)
  791.             auth = 'Basic %s' % base64.b64encode(raw).strip()
  792.             if req.headers.get(self.auth_header, None) == auth:
  793.                 return None
  794.             
  795.             req.add_header(self.auth_header, auth)
  796.             return self.parent.open(req)
  797.         else:
  798.             return None
  799.  
  800.  
  801.  
  802. class HTTPBasicAuthHandler(AbstractBasicAuthHandler, BaseHandler):
  803.     auth_header = 'Authorization'
  804.     
  805.     def http_error_401(self, req, fp, code, msg, headers):
  806.         url = req.get_full_url()
  807.         return self.http_error_auth_reqed('www-authenticate', url, req, headers)
  808.  
  809.  
  810.  
  811. class ProxyBasicAuthHandler(AbstractBasicAuthHandler, BaseHandler):
  812.     auth_header = 'Proxy-authorization'
  813.     
  814.     def http_error_407(self, req, fp, code, msg, headers):
  815.         authority = req.get_host()
  816.         return self.http_error_auth_reqed('proxy-authenticate', authority, req, headers)
  817.  
  818.  
  819.  
  820. def randombytes(n):
  821.     '''Return n random bytes.'''
  822.     pass
  823.  
  824.  
  825. class AbstractDigestAuthHandler:
  826.     
  827.     def __init__(self, passwd = None):
  828.         if passwd is None:
  829.             passwd = HTTPPasswordMgr()
  830.         
  831.         self.passwd = passwd
  832.         self.add_password = self.passwd.add_password
  833.         self.retried = 0
  834.         self.nonce_count = 0
  835.  
  836.     
  837.     def reset_retry_count(self):
  838.         self.retried = 0
  839.  
  840.     
  841.     def http_error_auth_reqed(self, auth_header, host, req, headers):
  842.         authreq = headers.get(auth_header, None)
  843.         if authreq:
  844.             scheme = authreq.split()[0]
  845.             if scheme.lower() == 'digest':
  846.                 return self.retry_http_digest_auth(req, authreq)
  847.             
  848.         
  849.  
  850.     
  851.     def retry_http_digest_auth(self, req, auth):
  852.         (token, challenge) = auth.split(' ', 1)
  853.         chal = parse_keqv_list(parse_http_list(challenge))
  854.         auth = self.get_authorization(req, chal)
  855.         if auth:
  856.             auth_val = 'Digest %s' % auth
  857.             if req.headers.get(self.auth_header, None) == auth_val:
  858.                 return None
  859.             
  860.             req.add_unredirected_header(self.auth_header, auth_val)
  861.             resp = self.parent.open(req)
  862.             return resp
  863.         
  864.  
  865.     
  866.     def get_cnonce(self, nonce):
  867.         dig = hashlib.sha1('%s:%s:%s:%s' % (self.nonce_count, nonce, time.ctime(), randombytes(8))).hexdigest()
  868.         return dig[:16]
  869.  
  870.     
  871.     def get_authorization(self, req, chal):
  872.         
  873.         try:
  874.             realm = chal['realm']
  875.             nonce = chal['nonce']
  876.             qop = chal.get('qop')
  877.             algorithm = chal.get('algorithm', 'MD5')
  878.             opaque = chal.get('opaque', None)
  879.         except KeyError:
  880.             return None
  881.  
  882.         (H, KD) = self.get_algorithm_impls(algorithm)
  883.         if H is None:
  884.             return None
  885.         
  886.         (user, pw) = self.passwd.find_user_password(realm, req.get_full_url())
  887.         if user is None:
  888.             return None
  889.         
  890.         if req.has_data():
  891.             entdig = self.get_entity_digest(req.get_data(), chal)
  892.         else:
  893.             entdig = None
  894.         A1 = '%s:%s:%s' % (user, realm, pw)
  895.         A2 = '%s:%s' % (req.get_method(), req.get_selector())
  896.         if qop == 'auth':
  897.             self.nonce_count += 1
  898.             ncvalue = '%08x' % self.nonce_count
  899.             cnonce = self.get_cnonce(nonce)
  900.             noncebit = '%s:%s:%s:%s:%s' % (nonce, ncvalue, cnonce, qop, H(A2))
  901.             respdig = KD(H(A1), noncebit)
  902.         elif qop is None:
  903.             respdig = KD(H(A1), '%s:%s' % (nonce, H(A2)))
  904.         else:
  905.             raise URLError("qop '%s' is not supported." % qop)
  906.         base = 'username="%s", realm="%s", nonce="%s", uri="%s", response="%s"' % (user, realm, nonce, req.get_selector(), respdig)
  907.         if opaque:
  908.             base += ', opaque="%s"' % opaque
  909.         
  910.         if entdig:
  911.             base += ', digest="%s"' % entdig
  912.         
  913.         base += ', algorithm="%s"' % algorithm
  914.         if qop:
  915.             base += ', qop=auth, nc=%s, cnonce="%s"' % (ncvalue, cnonce)
  916.         
  917.         return base
  918.  
  919.     
  920.     def get_algorithm_impls(self, algorithm):
  921.         if algorithm == 'MD5':
  922.             
  923.             H = lambda x: hashlib.md5(x).hexdigest()
  924.         elif algorithm == 'SHA':
  925.             
  926.             H = lambda x: hashlib.sha1(x).hexdigest()
  927.         
  928.         
  929.         KD = lambda s, d: H('%s:%s' % (s, d))
  930.         return (H, KD)
  931.  
  932.     
  933.     def get_entity_digest(self, data, chal):
  934.         pass
  935.  
  936.  
  937.  
  938. class HTTPDigestAuthHandler(BaseHandler, AbstractDigestAuthHandler):
  939.     '''An authentication protocol defined by RFC 2069
  940.  
  941.     Digest authentication improves on basic authentication because it
  942.     does not transmit passwords in the clear.
  943.     '''
  944.     auth_header = 'Authorization'
  945.     handler_order = 490
  946.     
  947.     def http_error_401(self, req, fp, code, msg, headers):
  948.         host = urlparse.urlparse(req.get_full_url())[1]
  949.         retry = self.http_error_auth_reqed('www-authenticate', host, req, headers)
  950.         self.reset_retry_count()
  951.         return retry
  952.  
  953.  
  954.  
  955. class ProxyDigestAuthHandler(BaseHandler, AbstractDigestAuthHandler):
  956.     auth_header = 'Proxy-Authorization'
  957.     handler_order = 490
  958.     
  959.     def http_error_407(self, req, fp, code, msg, headers):
  960.         host = req.get_host()
  961.         retry = self.http_error_auth_reqed('proxy-authenticate', host, req, headers)
  962.         self.reset_retry_count()
  963.         return retry
  964.  
  965.  
  966.  
  967. class AbstractHTTPHandler(BaseHandler):
  968.     
  969.     def __init__(self, debuglevel = 0):
  970.         self._debuglevel = debuglevel
  971.  
  972.     
  973.     def set_http_debuglevel(self, level):
  974.         self._debuglevel = level
  975.  
  976.     
  977.     def do_request_(self, request):
  978.         host = request.get_host()
  979.         if not host:
  980.             raise URLError('no host given')
  981.         
  982.         if request.has_data():
  983.             data = request.get_data()
  984.             if not request.has_header('Content-type'):
  985.                 request.add_unredirected_header('Content-type', 'application/x-www-form-urlencoded')
  986.             
  987.             if not request.has_header('Content-length'):
  988.                 request.add_unredirected_header('Content-length', '%d' % len(data))
  989.             
  990.         
  991.         (scheme, sel) = splittype(request.get_selector())
  992.         (sel_host, sel_path) = splithost(sel)
  993.         if not request.has_header('Host'):
  994.             if not sel_host:
  995.                 pass
  996.             request.add_unredirected_header('Host', host)
  997.         
  998.         for name, value in self.parent.addheaders:
  999.             name = name.capitalize()
  1000.             if not request.has_header(name):
  1001.                 request.add_unredirected_header(name, value)
  1002.                 continue
  1003.         
  1004.         return request
  1005.  
  1006.     
  1007.     def do_open(self, http_class, req):
  1008.         '''Return an addinfourl object for the request, using http_class.
  1009.  
  1010.         http_class must implement the HTTPConnection API from httplib.
  1011.         The addinfourl return value is a file-like object.  It also
  1012.         has methods and attributes including:
  1013.             - info(): return a mimetools.Message object for the headers
  1014.             - geturl(): return the original request URL
  1015.             - code: HTTP status code
  1016.         '''
  1017.         host = req.get_host()
  1018.         if not host:
  1019.             raise URLError('no host given')
  1020.         
  1021.         h = http_class(host)
  1022.         h.set_debuglevel(self._debuglevel)
  1023.         headers = dict(req.headers)
  1024.         headers.update(req.unredirected_hdrs)
  1025.         headers['Connection'] = 'close'
  1026.         headers = dict((lambda .0: for name, val in .0:
  1027. (name.title(), val))(headers.items()))
  1028.         
  1029.         try:
  1030.             h.request(req.get_method(), req.get_selector(), req.data, headers)
  1031.             r = h.getresponse()
  1032.         except socket.error:
  1033.             err = None
  1034.             raise URLError(err)
  1035.  
  1036.         r.recv = r.read
  1037.         fp = socket._fileobject(r, close = True)
  1038.         resp = addinfourl(fp, r.msg, req.get_full_url())
  1039.         resp.code = r.status
  1040.         resp.msg = r.reason
  1041.         return resp
  1042.  
  1043.  
  1044.  
  1045. class HTTPHandler(AbstractHTTPHandler):
  1046.     
  1047.     def http_open(self, req):
  1048.         return self.do_open(httplib.HTTPConnection, req)
  1049.  
  1050.     http_request = AbstractHTTPHandler.do_request_
  1051.  
  1052. if hasattr(httplib, 'HTTPS'):
  1053.     
  1054.     class HTTPSHandler(AbstractHTTPHandler):
  1055.         
  1056.         def https_open(self, req):
  1057.             return self.do_open(httplib.HTTPSConnection, req)
  1058.  
  1059.         https_request = AbstractHTTPHandler.do_request_
  1060.  
  1061.  
  1062.  
  1063. class HTTPCookieProcessor(BaseHandler):
  1064.     
  1065.     def __init__(self, cookiejar = None):
  1066.         import cookielib as cookielib
  1067.         if cookiejar is None:
  1068.             cookiejar = cookielib.CookieJar()
  1069.         
  1070.         self.cookiejar = cookiejar
  1071.  
  1072.     
  1073.     def http_request(self, request):
  1074.         self.cookiejar.add_cookie_header(request)
  1075.         return request
  1076.  
  1077.     
  1078.     def http_response(self, request, response):
  1079.         self.cookiejar.extract_cookies(response, request)
  1080.         return response
  1081.  
  1082.     https_request = http_request
  1083.     https_response = http_response
  1084.  
  1085.  
  1086. class UnknownHandler(BaseHandler):
  1087.     
  1088.     def unknown_open(self, req):
  1089.         type = req.get_type()
  1090.         raise URLError('unknown url type: %s' % type)
  1091.  
  1092.  
  1093.  
  1094. def parse_keqv_list(l):
  1095.     '''Parse list of key=value strings where keys are not duplicated.'''
  1096.     parsed = { }
  1097.     for elt in l:
  1098.         (k, v) = elt.split('=', 1)
  1099.         if v[0] == '"' and v[-1] == '"':
  1100.             v = v[1:-1]
  1101.         
  1102.         parsed[k] = v
  1103.     
  1104.     return parsed
  1105.  
  1106.  
  1107. def parse_http_list(s):
  1108.     '''Parse lists as described by RFC 2068 Section 2.
  1109.  
  1110.     In particular, parse comma-separated lists where the elements of
  1111.     the list may include quoted-strings.  A quoted-string could
  1112.     contain a comma.  A non-quoted string could have quotes in the
  1113.     middle.  Neither commas nor quotes count if they are escaped.
  1114.     Only double-quotes count, not single-quotes.
  1115.     '''
  1116.     res = []
  1117.     part = ''
  1118.     escape = quote = False
  1119.     for cur in s:
  1120.         if escape:
  1121.             part += cur
  1122.             escape = False
  1123.             continue
  1124.         
  1125.         if quote:
  1126.             if cur == '\\':
  1127.                 escape = True
  1128.                 continue
  1129.             elif cur == '"':
  1130.                 quote = False
  1131.             
  1132.             part += cur
  1133.             continue
  1134.         
  1135.         if cur == ',':
  1136.             res.append(part)
  1137.             part = ''
  1138.             continue
  1139.         
  1140.         if cur == '"':
  1141.             quote = True
  1142.         
  1143.         part += cur
  1144.     
  1145.     if part:
  1146.         res.append(part)
  1147.     
  1148.     return [ part.strip() for part in res ]
  1149.  
  1150.  
  1151. class FileHandler(BaseHandler):
  1152.     
  1153.     def file_open(self, req):
  1154.         url = req.get_selector()
  1155.         if url[:2] == '//' and url[2:3] != '/':
  1156.             req.type = 'ftp'
  1157.             return self.parent.open(req)
  1158.         else:
  1159.             return self.open_local_file(req)
  1160.  
  1161.     names = None
  1162.     
  1163.     def get_names(self):
  1164.         if FileHandler.names is None:
  1165.             
  1166.             try:
  1167.                 FileHandler.names = (socket.gethostbyname('localhost'), socket.gethostbyname(socket.gethostname()))
  1168.             except socket.gaierror:
  1169.                 FileHandler.names = (socket.gethostbyname('localhost'),)
  1170.             except:
  1171.                 None<EXCEPTION MATCH>socket.gaierror
  1172.             
  1173.  
  1174.         None<EXCEPTION MATCH>socket.gaierror
  1175.         return FileHandler.names
  1176.  
  1177.     
  1178.     def open_local_file(self, req):
  1179.         import email.Utils as email
  1180.         import mimetypes as mimetypes
  1181.         host = req.get_host()
  1182.         file = req.get_selector()
  1183.         localfile = url2pathname(file)
  1184.         stats = os.stat(localfile)
  1185.         size = stats.st_size
  1186.         modified = email.Utils.formatdate(stats.st_mtime, usegmt = True)
  1187.         mtype = mimetypes.guess_type(file)[0]
  1188.         if not mtype:
  1189.             pass
  1190.         headers = mimetools.Message(StringIO('Content-type: %s\nContent-length: %d\nLast-modified: %s\n' % ('text/plain', size, modified)))
  1191.         if host:
  1192.             (host, port) = splitport(host)
  1193.         
  1194.         if (not host or not port) and socket.gethostbyname(host) in self.get_names():
  1195.             return addinfourl(open(localfile, 'rb'), headers, 'file:' + file)
  1196.         
  1197.         raise URLError('file not on local host')
  1198.  
  1199.  
  1200.  
  1201. class FTPHandler(BaseHandler):
  1202.     
  1203.     def ftp_open(self, req):
  1204.         import ftplib as ftplib
  1205.         import mimetypes
  1206.         host = req.get_host()
  1207.         if not host:
  1208.             raise IOError, ('ftp error', 'no host given')
  1209.         
  1210.         (host, port) = splitport(host)
  1211.         if port is None:
  1212.             port = ftplib.FTP_PORT
  1213.         else:
  1214.             port = int(port)
  1215.         (user, host) = splituser(host)
  1216.         if user:
  1217.             (user, passwd) = splitpasswd(user)
  1218.         else:
  1219.             passwd = None
  1220.         host = unquote(host)
  1221.         if not user:
  1222.             pass
  1223.         user = unquote('')
  1224.         if not passwd:
  1225.             pass
  1226.         passwd = unquote('')
  1227.         
  1228.         try:
  1229.             host = socket.gethostbyname(host)
  1230.         except socket.error:
  1231.             msg = None
  1232.             raise URLError(msg)
  1233.  
  1234.         (path, attrs) = splitattr(req.get_selector())
  1235.         dirs = path.split('/')
  1236.         dirs = map(unquote, dirs)
  1237.         dirs = dirs[:-1]
  1238.         file = dirs[-1]
  1239.         if dirs and not dirs[0]:
  1240.             dirs = dirs[1:]
  1241.         
  1242.         
  1243.         try:
  1244.             fw = self.connect_ftp(user, passwd, host, port, dirs)
  1245.             if not file or 'I':
  1246.                 pass
  1247.             type = 'D'
  1248.             for attr in attrs:
  1249.                 (attr, value) = splitvalue(attr)
  1250.                 if attr.lower() == 'type' and value in ('a', 'A', 'i', 'I', 'd', 'D'):
  1251.                     type = value.upper()
  1252.                     continue
  1253.             
  1254.             (fp, retrlen) = fw.retrfile(file, type)
  1255.             headers = ''
  1256.             mtype = mimetypes.guess_type(req.get_full_url())[0]
  1257.             if mtype:
  1258.                 headers += 'Content-type: %s\n' % mtype
  1259.             
  1260.             if retrlen is not None and retrlen >= 0:
  1261.                 headers += 'Content-length: %d\n' % retrlen
  1262.             
  1263.             sf = StringIO(headers)
  1264.             headers = mimetools.Message(sf)
  1265.             return addinfourl(fp, headers, req.get_full_url())
  1266.         except ftplib.all_errors:
  1267.             msg = None
  1268.             raise IOError, ('ftp error', msg), sys.exc_info()[2]
  1269.  
  1270.  
  1271.     
  1272.     def connect_ftp(self, user, passwd, host, port, dirs):
  1273.         fw = ftpwrapper(user, passwd, host, port, dirs)
  1274.         return fw
  1275.  
  1276.  
  1277.  
  1278. class CacheFTPHandler(FTPHandler):
  1279.     
  1280.     def __init__(self):
  1281.         self.cache = { }
  1282.         self.timeout = { }
  1283.         self.soonest = 0
  1284.         self.delay = 60
  1285.         self.max_conns = 16
  1286.  
  1287.     
  1288.     def setTimeout(self, t):
  1289.         self.delay = t
  1290.  
  1291.     
  1292.     def setMaxConns(self, m):
  1293.         self.max_conns = m
  1294.  
  1295.     
  1296.     def connect_ftp(self, user, passwd, host, port, dirs):
  1297.         key = (user, host, port, '/'.join(dirs))
  1298.         if key in self.cache:
  1299.             self.timeout[key] = time.time() + self.delay
  1300.         else:
  1301.             self.cache[key] = ftpwrapper(user, passwd, host, port, dirs)
  1302.             self.timeout[key] = time.time() + self.delay
  1303.         self.check_cache()
  1304.         return self.cache[key]
  1305.  
  1306.     
  1307.     def check_cache(self):
  1308.         t = time.time()
  1309.         if self.soonest <= t:
  1310.             for k, v in self.timeout.items():
  1311.                 if v < t:
  1312.                     self.cache[k].close()
  1313.                     del self.cache[k]
  1314.                     del self.timeout[k]
  1315.                     continue
  1316.             
  1317.         
  1318.         self.soonest = min(self.timeout.values())
  1319.         if len(self.cache) == self.max_conns:
  1320.             for k, v in self.timeout.items():
  1321.                 if v == self.soonest:
  1322.                     del self.cache[k]
  1323.                     del self.timeout[k]
  1324.                     break
  1325.                     continue
  1326.             
  1327.             self.soonest = min(self.timeout.values())
  1328.         
  1329.  
  1330.  
  1331.  
  1332. class GopherHandler(BaseHandler):
  1333.     
  1334.     def gopher_open(self, req):
  1335.         import gopherlib as gopherlib
  1336.         host = req.get_host()
  1337.         if not host:
  1338.             raise GopherError('no host given')
  1339.         
  1340.         host = unquote(host)
  1341.         selector = req.get_selector()
  1342.         (type, selector) = splitgophertype(selector)
  1343.         (selector, query) = splitquery(selector)
  1344.         selector = unquote(selector)
  1345.         if query:
  1346.             query = unquote(query)
  1347.             fp = gopherlib.send_query(selector, query, host)
  1348.         else:
  1349.             fp = gopherlib.send_selector(selector, host)
  1350.         return addinfourl(fp, noheaders(), req.get_full_url())
  1351.  
  1352.  
  1353.